package com.itheima.reggie.controller;

import com.itheima.reggie.common.R;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.UUID;

@RestController
public class FileController {
    
    @Value("${reggie.file-path}")
    private String path;
    
    /**
     * 文件上传
     *      1. 接收页面传递上来的文件
     *      2. 保存到硬盘上
     *      3. 处理细节：
     *          3.1 保存文件的时候，目录不会自动创建，所以要确保这个目录必须存在，可以使用代码来创建目录
     *          3.2 必须确保文件的名称唯一 , 可以使用UUID... 生成一串随机字符串
     *          3.3 文件的后缀名必须是动态变化，不能固定是 jpg | png ... 可以获取原始文件名，然后切割出来后缀名
     *          3.4 上传成功之后，要给页面返回文件的名字，以便后续的下载操作，可以下载这张图片。
     * @param file
     * @return
     */
    @PostMapping("/common/upload")
    public R upload(MultipartFile file){
    
        try {
            //1. 获取原始文件名 , 切割出来后缀
            String filename = file.getOriginalFilename(); //   aa.jpg
            String suffix = filename.substring(filename.lastIndexOf('.'));  //   .jpg
        
            //2. 构建一个新的文件名
            String newName = UUID.randomUUID().toString().replace("-","") + suffix; //   69dd8caa24d641d28e0dcc313ad5d32a.jpg
        
            //3. 准备好保存的目录，确保这个目录是存在的
            File dir = new File(path);
            if(!dir.exists()){ //如果目录不存在
               dir.mkdirs();
            }
        
            //4. 保存文件
            File f = new File(path+newName);  // D:/heima123/69dd8caa24d641d28e0dcc313ad5d32a.jpg
            file.transferTo(f);
        
            //5. 返回结果（文件的名字）给页面
            return R.success(newName);
        } catch (IOException e) {
            e.printStackTrace();
            
            return R.error("上传失败！");
        }
    }
    
    
    /**
     * 文件下载
     *      1. 获取要下载的文件
     *      2. 到以前保存文件的目录里面去找出来这个文件
     *      3. 然后使用IO对接，把文件写到页面上
     *      4. 最后可以给一个结果 ： 成功，失败！
     * @param name
     * @return
     */
    @GetMapping("/common/download")
    public R download(String name , HttpServletResponse response){ // aa.jpg
    
        try {
            
            //1. 定位文件的位置
            File file = new File(path +  name);
        
            //2. 把这个文件读到缓冲区里面去。
            FileInputStream fis = new FileInputStream(file);
            
            //2.1. 获取输出流对象
            ServletOutputStream os = response.getOutputStream();
            
//            IOUtils.copy(fis , os);
    
            //2.2 定义缓冲区
            byte [] buffer = new byte[1024];
            
            //2.3 读到缓冲区
            int len = 0 ;
            while( (len = fis.read(buffer)) != -1 ){
                
                //2.4 输入流和输出对接
                os.write(buffer , 0 , len);
                os.flush();
            }
            
            //2.4 收尾
            os.close();
            fis.close();
            
            //3. 响应
            return R.success("下载成功");
            
        } catch (Exception e) {
            e.printStackTrace();
            return R.error("下载失败");
        }
    
    }
    
}
